#!/bin/bash

# 设置颜色
GREEN='\033[0;32m'
RED='\033[0;31m'
NC='\033[0m'

# 检查虚拟环境
if [ ! -d "venv" ]; then
    echo -e "${RED}错误: 未找到虚拟环境，请先运行 install.sh${NC}"
    exit 1
fi

# 检查配置文件
if [ ! -f ".env" ]; then
    echo -e "${RED}警告: 未找到 .env 文件，将使用默认配置${NC}"
    cp .env.example .env
fi

# 创建必要的目录
mkdir -p logs
mkdir -p chat_sessions

# 激活虚拟环境
source venv/bin/activate

# 检查 Ollama 服务
if ! curl -s http://localhost:11434/api/version > /dev/null; then
    echo -e "${RED}错误: Ollama 服务未运行${NC}"
    echo "请先启动 Ollama 服务:"
    echo "ollama serve"
    exit 1
fi

# 运行应用
echo -e "${GREEN}启动 Ollama Chat...${NC}"
python -m streamlit run chat_app.py "$@" 